package org.example.matrix;

import java.util.ArrayList;
import java.util.List;

public class SpiralOrder {
    /**
     * 给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素
     * 示例 1：
     *
     *
     * 输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
     * 输出：[1,2,3,6,9,8,7,4,5]
     * @param matrix
     * @return
     */
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> ret = new ArrayList<>();
        int m = matrix.length,n = matrix[0].length;
        boolean[][] visited = new boolean[m][n];
        int size = m * n;
        int row = 0,col = 0;
        //方向数组
        int[][] direction = {{0,1},{1,0},{0,-1},{-1,0}};
        //开始向右走
        int directionIndex = 0;
        //循环所有元素结束
        while(size--!=0){
            ret.add(matrix[row][col]);
            visited[row][col] = true;
            //计算下一个位置
            int nextRow = row + direction[directionIndex][0];
            int nextCol = col + direction[directionIndex][1];
            //判断是否到达边界
            if(nextRow < 0 || nextRow >= m || nextCol < 0 || nextCol >= n ||
                    visited[nextRow][nextCol]){
                //顺时针转向
                directionIndex = (directionIndex+1)%4;
            }
            //最终得出下一个位置
            row = row + direction[directionIndex][0];
            col = col + direction[directionIndex][1];
        }
        return ret;
    }
}
